import os
import shutil

"""
数据集搬运与移动
"""
############需要修改###################
# 指定源文件夹和目标文件夹路径
images_source_dir = "/home/hw/dataset/cone_total_new/train/images"    # 图片文件夹路径
images_target_dir = "/home/hw/dataset/cone_total_new/images"        # 图片目标路径
labels_source_dir = "/home/hw/dataset/cone_total_new/train/labels"    # 标签文件夹路径
labels_target_dir = "/home/hw/dataset/cone_total_new/labels"  
#####################################

def move_files(source_dir, target_dir):
    # 如果目标文件夹不存在，则创建
    if not os.path.exists(target_dir):
        os.makedirs(target_dir)
    
    # 遍历源文件夹下的所有文件和子文件夹
    for root, dirs, files in os.walk(source_dir):
        for file in files:
            # 构建源文件的完整路径
            source_file_path = os.path.join(root, file)
            
            # 构建目标文件的完整路径
            target_file_path = os.path.join(target_dir, file)
            
            # 移动文件
            shutil.move(source_file_path, target_file_path)
            print(f"Moved {source_file_path} to {target_file_path}")

      # 标签目标路径

# 移动图片文件夹下的内容到指定路径
move_files(images_source_dir, images_target_dir)

# 移动标签文件夹下的内容到指定路径
move_files(labels_source_dir, labels_target_dir)